from __future__ import division
from model.yolo_util import detect, load_classes
import yaml

_, CLASSES = load_classes('./cfg/classes.names')

if __name__ == '__main__':
    with open('cfg/cfg.yaml', 'r') as loadfile:
        config = yaml.load_all(loadfile, Loader=yaml.FullLoader)
        config_all = [x for x in config]

    # eval mode
    config = config_all[1]

    detect(config, CLASSES)
